#!/usr/bin/env python
#--coding:utf-8--
'''
初始化数据库
'''
import sys
import os

sys.path.append(os.path.dirname(os.path.abspath(__file__)))
from 主机管理.models import SQLClass
from 主机管理.models import loginMessage
HD=None

def createTable():
    global HD
    HD=SQLClass.SQLBase()
    HD.drop_all()
    HD.create_all()
    HD.addTable(SQLClass.Users)
    HD.add(username="alex",password=loginMessage.mkpasswd("123456"))
    HD.add(username="jack", password=loginMessage.mkpasswd("456456"))
    HD.add(username="roro", password=loginMessage.mkpasswd("asdasd"))
    HD.addTable(SQLClass.Groups)
    HD.add(groupname="haproxy_group")
    HD.add(groupname="web_group")
    HD.add(groupname="mysql_group")
    HD.add(groupname="nagios_group")
    group_dict = {}
    for i in HD.watch():
        group_dict.setdefault(i.groupname,i.groupid)
    HD.addTable(SQLClass.Hosts)
    #hostid=%s,hostname=%s,ipattr=%s,groupid=%s

    HD.add(hostname="haproxy",ipattr="10.0.0.40",port=22,groupid=group_dict.get("haproxy_group"))
    HD.add(hostname="lamp", ipattr="10.0.0.41",port=22,groupid=group_dict.get("web_group"))
    HD.commit()


'''
if __name__=="__main__":
    createTable()
'''